package online.longzipeng.mywebdemo.mq.consumer;

import cn.hutool.json.JSONObject;
import cn.hutool.json.JSONUtil;
import com.rabbitmq.client.Channel;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.annotation.Queue;
import org.springframework.amqp.rabbit.annotation.RabbitListener;
import org.springframework.stereotype.Component;

import java.io.IOException;

/**
 * @Author: lzp
 * @description: 工作模式，多消费者
 * 与简单模式的结构一致，一个队列两端，只不过多了几个消费者
 * @Date: 2024/1/14
 */
@Component
public class WorkConsumer {

	@RabbitListener(queuesToDeclare = @Queue("work_queue"))
	public void process1(Message message, Channel channel) throws IOException {
		String str = new String(message.getBody());
		JSONObject msgData = JSONUtil.parseObj(str);
		Object messageId = msgData.getByPath("messageId");
		if (null == messageId || messageId.toString().equals("")) {
			/**
			 * 有异常就拒收消息
			 * basicNack(long deliveryTag, boolean multiple, boolean requeue)
			 * deliveryTag:当前消息在队列中的的索引;
			 * multiple:为true的话就是批量确认 【deliveryTag小于当前消息的全部确认】
			 * requeue:true将消息重返当前消息队列,重新发送给消费者;
			 *         false将消息丢弃
			 */
			channel.basicNack(message.getMessageProperties().getDeliveryTag(), false, true);
			System.out.println("工作模式1 -> 消费者拒收消息 : " + msgData.toString());
		} else {
			/**
			 * 没有异常就确认消息
			 * basicAck(long deliveryTag, boolean multiple)
			 * deliveryTag:当前消息在队列中的的索引;
			 * multiple:为true的话就是批量确认
			 */
			channel.basicAck(message.getMessageProperties().getDeliveryTag(), false);
			System.out.println("工作模式1 -> 消费者收到消息  : " + msgData.toString());
		}
	}


	@RabbitListener(queuesToDeclare = @Queue("work_queue"))
	public void process2(Message message, Channel channel) throws IOException {
		String str = new String(message.getBody());
		JSONObject msgData = JSONUtil.parseObj(str);
		Object messageId = msgData.getByPath("messageId");
		if (null == messageId || messageId.toString().equals("")) {
			/**
			 * 有异常就拒收消息
			 * basicNack(long deliveryTag, boolean multiple, boolean requeue)
			 * deliveryTag:当前消息在队列中的的索引;
			 * multiple:为true的话就是批量确认 【deliveryTag小于当前消息的全部确认】
			 * requeue:true将消息重返当前消息队列,重新发送给消费者;
			 *         false将消息丢弃
			 */
			channel.basicNack(message.getMessageProperties().getDeliveryTag(), false, true);
			System.out.println("工作模式2 -> 消费者拒收消息 : " + msgData.toString());
		} else {
			/**
			 * 没有异常就确认消息
			 * basicAck(long deliveryTag, boolean multiple)
			 * deliveryTag:当前消息在队列中的的索引;
			 * multiple:为true的话就是批量确认
			 */
			channel.basicAck(message.getMessageProperties().getDeliveryTag(), false);
			System.out.println("工作模式2 -> 消费者收到消息  : " + msgData.toString());
		}
	}

}
